写测试写到一定程度,就会发现,不做mock是不行的, 一个长达5个小时的计算代码,总不能每次测试都跑一次吧?

这个时候我们就需要mock住代码

所谓mock,简单理解就是模拟代码的行为, 并不真正运行代码本身,从而能够减少重复测试,减少测试的时间.

python自带的单元测试库是unittest,有对应的unittest.mock 可以来进行mock.

pytest对应的mock就是pytest-mock module.

我个人更加喜欢使用pytest, pytest里的mocker也相对简单一些, 大部分unittest.mock的接口, mocker都是支持的.

比如mocker.patch() 和mock 的patch接口是一样的, 参考 https://docs.python.org/3/library/unittest.mock.html#patch

　　{mock}.assert_not_called()

　　{mock}.assert_called()

　　{mock}.assert_called_with()

　　{mock}.assert_called_once()

　　{mock}.assert_called_once_with()

　　{mock}.assert_has_calls()

　　{mock}.assert_any_call()

注意你引入的mock的module不一定要是一个第三方包,也可以是你自己的python代码, 比如mypython.py, 使用 'import mypython'

这样就可以测试你自己写的代码.

patch的必须是一个已经存在的东西,可以使用{module}.{function}来调用的这种.

如果想要mock你的py代码引用的module里的function,则需要在你的py代码里使用 'from module import function',在测试代码里就可以使用 mypython.function 来对其进行mock

测试代码 mypython_test.py

你的代码 mypython.py